import os
import torch


class Config:
    def __init__(self):
        self.project_name = "masac day-ahead schedule"
        self.entity = "yizhigopher"
        self.agent_name = "msac"
        self.use_wandb = False

        self.actor_lr = 3e-4
        self.critic_lr = 1e-3
        self.gamma = 0.99
        self.device = torch.device("cuda") if torch.cuda.is_available() \
            else torch.device("cpu")
        self.alpha_lr = 1e-3
        self.autotune = True
        self.alpha = 0.1
        self.buffer_size = 100000
        self.episode = 1000
        self.minial_size = 1000
        self.batch_size = 64
        self.tau = 0.005
        self.is_save_model = True
        self.root_path = os.path.join(
            os.path.dirname(__file__),
            "..",
        )
        self.model_path = self.root_path + \
            f"\\result\\%s.pth" % (self.agent_name)
